package utils;

import com.alibaba.excel.EasyExcel;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

/**
 * @Description: 导出Excel工具类
 * @Author: syh
 * @Date: 2023/12/06/11:15
 */
public class ExportExcelUtilsWithEasy {

    /**
     * 执行导出功能
     * @param fileName 生成文件的路径
     * @param dataList 数据列表
     * @param fieldList 字段列表
     */
    public static void doExport(String fileName, List<List<String>> dataList,List<String> fieldList){
        List<List<String>> newHeaderList = new ArrayList<>();
        for (String eleStr : fieldList) {
            ArrayList<String> eleArr = new ArrayList<>();
            eleArr.add(eleStr);
            newHeaderList.add(eleArr);
        }


        File excelFile = new File(fileName);
        if (excelFile.exists() && excelFile.isFile()) {
            excelFile.delete();
        }

        EasyExcel.write(fileName)
                // 这里放入动态头
                .head(newHeaderList).sheet("查询数据")
                // 当然这里数据也可以用 List<List<String>> 去传入
                .doWrite(dataList);

    }
}
